package cn.flexui.flexuiserver.controller.web;

import cn.dev33.satoken.stp.StpUtil;
import cn.flexui.flexuiserver.common.ResponseResultGenerator;
import cn.flexui.flexuiserver.common.convert.PromptTemplateConvert;
import cn.flexui.flexuiserver.domain.ResponseResult;
import cn.flexui.flexuiserver.domain.dto.PromptTemplateDTO;
import cn.flexui.flexuiserver.domain.entity.PromptTemplate;
import cn.flexui.flexuiserver.domain.vo.PromptTemplateVO;
import cn.flexui.flexuiserver.service.PromptTemplateService;
import io.swagger.v3.oas.annotations.Operation;
import io.swagger.v3.oas.annotations.tags.Tag;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.web.bind.annotation.GetMapping;
import org.springframework.web.bind.annotation.PostMapping;
import org.springframework.web.bind.annotation.RequestMapping;
import org.springframework.web.bind.annotation.RestController;

import java.util.List;

/**
 * <p>
 * 提示词模板 前端控制器
 * </p>
 *
 * @author apeto
 * @since 2024-09-17 07:08:21
 */
@Tag(name = "提示词模板")
@RestController
@RequestMapping("/promptTemplate")
public class PromptTemplateController {

    @Autowired
    private PromptTemplateService promptTemplateService;
    @Autowired
    private PromptTemplateConvert promptTemplateConvert;

    @Operation(summary = "保存或修改")
    @PostMapping("/saveOrUpdate")
    public ResponseResult<?> saveOrUpdate(PromptTemplateDTO promptTemplateDTO) {

        long userId = StpUtil.getLoginIdAsLong();
        PromptTemplate promptTemplate = new PromptTemplate();
        promptTemplate.setContent(promptTemplateDTO.getContent());
        promptTemplate.setId(promptTemplateDTO.getId());
        promptTemplate.setUserId(userId);
        promptTemplateService.saveOrUpdate(promptTemplate);
        return ResponseResultGenerator.success();
    }

    @Operation(summary = "列表")
    @GetMapping("/list")
    public ResponseResult<List<PromptTemplateVO>> list() {
        long userId = StpUtil.getLoginIdAsLong();
        List<PromptTemplate> list = promptTemplateService.selectByUserId(userId);
        return ResponseResultGenerator.success(promptTemplateConvert.toPromptTemplateListvVO(list));
    }

}
